import java.util.LinkedList;
import java.util.Queue;

class Solution04 {
    // 定义数轴
    int[] dx = { 0, 0, -1, 1 };
    int[] dy = { 1, -1, 0, 0 };
    int m, n;
    // 定义同等规模的数组
    int[][] ret;

    public int[][] highestPeak(int[][] isWater) {
        m = isWater.length;
        n = isWater[0].length;
        ret = new int[m][n];
        // 初始化结果数组
        for (int i = 0; i < m; i++) {
            for (int j = 0; j < n; j++) {
                ret[i][j] = -1;
            }
        }
        // 创建一个队列
        Queue<int[]> q = new LinkedList<>();
        for (int i = 0; i < m; i++) {
            for (int j = 0; j < n; j++) {
                if (isWater[i][j] == 1) {
                    ret[i][j] = 0;
                    // 添加到数组
                    q.add(new int[] { i, j });
                }
            }
        }
        while (!q.isEmpty()) {
            int size = q.size();
            while (size-- != 0) {
                int[] t = q.poll();
                int a = t[0], b = t[1];
                // 上下左右四个方向
                for (int i = 0; i < 4; i++) {
                    int x = a + dx[i], y = b + dy[i];
                    // 进行判断
                    if (x >= 0 && x < m && y >= 0 && y < n && isWater[x][y] != 1 && ret[x][y] == -1) {
                        ret[x][y] = ret[a][b] + 1;
                        // 加入队列
                        q.add(new int[] { x, y });
                    }
                }
            }
        }
        return ret;
    }
}